from namo.r1.r1_grounding import main, GRPOScriptArguments, GRPOConfig, ModelConfig
from trl import TrlParser


if __name__ == "__main__":
    parser = TrlParser((GRPOScriptArguments, GRPOConfig, ModelConfig))
    script_args, training_args, model_args = parser.parse_args_and_config()
    main(script_args, training_args, model_args)
